package leetcode.Hot100;

import java.util.Arrays;

/**
 * @author Cheng Jun
 * Description: 给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间，CPU 可以完成一个任务，或者处于待命状态。
 * <p>
 * 然而，两个 相同种类 的任务之间必须有长度为整数 n 的冷却时间，因此至少有连续 n 个单位时间内 CPU 在执行不同的任务，或者在待命状态。
 * <p>
 * 你需要计算完成所有任务所需要的 最短时间 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/task-scheduler
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @version 1.0
 * @date 2021/11/29 10:33
 * 二刷
 */
public class leastInterval {

    public static void main(String[] args) {

        System.out.println(leastInterval(new char[]{'A', 'A', 'A', 'B', 'B', 'B'}, 2));
        System.out.println(leastInterval(new char[]{'A', 'A', 'A', 'A', 'A', 'A', 'B', 'C', 'D', 'E', 'F', 'G'}, 2));
    }

    // 先捡好的吃
    static int leastInterval(char[] tasks, int n) {
        // 特殊情况
        if (n == 0) {
            return tasks.length;
        }
        int time = 0;
        int[] charNum = new int[26];
        for (char task : tasks) {
            charNum[task - 65] += 1;
        }
        // 根据 字母出现顺序 倒序排序
        Arrays.sort(charNum);
        for (int i = 0; i < charNum.length; i++) {
            // 该位置任务被消耗完
            if (charNum[i] == 0) {
                continue;
            } else {
                // 消耗该位置一个任务
                charNum[i] -= 1;
                time++;
                for (int j = i + 1, counter = 0; j < charNum.length && counter < n; j++) {
                    if (charNum[j] != 0) {
                        charNum[j] -= 1;
                        time++;
                        counter++;
                    } else {
                        if (j == charNum.length - 1) {
                            if (charNum[i] != 0) {
                                time += n - counter;
                            }
                        }
                    }
                }
                i--;
            }
        }
        return time;
    }

    // 仔细想想
    int leastInterval1(char[] tasks, int n) {
        int count[] = new int[26];
        for (int i = 0; i < tasks.length; i++) {
            count[tasks[i] - 'A']++;
        }
        Arrays.sort(count);
        int j = 25;
        while (j >= 0 && count[j] == count[25]) {
            j--;
        }
        //从j+1到25并列最多
        return Math.max(tasks.length, (n + 1) * (count[25] - 1) + 25 - j);
    }
}
